from langchain.schema import HumanMessage, SystemMessage
from langchain.chat_models.gigachat import GigaChat
import os
from dotenv import load_dotenv

load_dotenv()

TOKEN = os.getenv("TOKEN_G")

class Chat:
    def __init__(self):
        self.chat = GigaChat(
        credentials=TOKEN, 
        verify_ssl_certs=False)
        
        self.messages = [
            SystemMessage("Ты системный администратор, который составляет акт по внедрению ПО для компании УрТК НИЯУ МИФИ: акт должен содержать описание ПО, характеристки ПО,возможности ПО и вывод по внедрению")
        ]
        
    def send_message(self, message):
        self.messages.append(HumanMessage(content=message))
        return self.send_chat()
    
    def send_chat(self):
        res = self.chat(self.messages) 
        return res.content   
    
    
        
        